package com.example.yian.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.example.yian.entity.SecurityRisk;
import org.apache.ibatis.annotations.Mapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.apache.ibatis.annotations.Param;

import java.sql.Timestamp;
import java.util.List;
import java.util.Map;

@Mapper
public interface SecurityRiskMapper extends BaseMapper<SecurityRisk> {

    /**
     * 根据user_id分页查询隐患列表
     *
     * @param page    分页对象
     * @param user_id 用户ID
     * @return 分页数据
     */
    List<SecurityRisk>  getRiskListByUserIdWithPagination(@Param("page") Page<SecurityRisk> page,
                                                          @Param("userId") Long userId,
                                                          @Param("startTime") String startTime,
                                                          @Param("endTime") String endTime,
                                                          @Param("limit") int limit,
                                                          @Param("offset") int offset,
                                                          @Param("selectFlag") int selectFlag);

    List<SecurityRisk>  getRiskListByUserId( @Param("userId") Long userId);


    /**
     * 根据用户ID和日期范围统计每日风险数量
     *
     * @param userId    用户ID
     * @param startDate 统计开始日期
     * @param endDate   统计结束日期
     * @return 每日风险数量映射
     */
    List<Map<String, Object>> countRisksByUserIdAndDateRange(@Param("userId") Long userId,
                                                             @Param("startDate") Timestamp startDate,
                                                             @Param("endDate") Timestamp endDate);
}